FROM swr.cn-north-4.myhuaweicloud.com/modelarts-job-dev-image/mindspore-ascend910-cp37-euleros2.8-aarch64-training:1.3.0-3.3.0-roma

USER root

ARG TMP_SPACE="/tmp"

COPY model_zoo $TMP_SPACE/model_zoo
RUN mkdir -p /home/ascend/mindspore && mv $TMP_SPACE/model_zoo /home/ascend/mindspore

#安装python依赖
RUN pip install seaborn --trusted-host https://repo.huaweicloud.com -i https://repo.huaweicloud.com/repository/pypi/simple

# 安装ATC工具
ARG ATC_PKG=Ascend-atc-*-linux.*.run
COPY toolkit/$ATC_PKG $TMP_SPACE
RUN chmod +x $TMP_SPACE/${ATC_PKG}  && $TMP_SPACE/$ATC_PKG --full && rm $TMP_SPACE/$ATC_PKG

ENV MINDSPORE_CODE_PATH=/home/ascend/mindspore/model_zoo/official/cv
ENV PATH=/usr/local/Ascend/atc/bin:$PATH
